from xlrd import open_workbook


# 数据驱动，通过这种方式  可以提高测试的复用性，减少重复测试的次数
class ReadExcel:
    def get_excel_data(self, ExcelName, sheet):
        # 打开.xls文件
        wb = open_workbook(f"../woniuExcel/{ExcelName}.xls")
        # 选择工作表
        ws = wb.sheet_by_index(sheet)
        # 获取最大行列数
        rows = ws.nrows
        cols = ws.ncols

        info_list = []

        for r in range(1, rows):
            # 定义一个列表 用来保存一个的信息
            lines = []
            for c in range(1, cols):
                lines.append(ws.cell_value(r, c))
            info_list.append(lines)

        return info_list


if __name__ == '__main__':
    r = ReadExcel()
    e = r.get_excel_data("蜗牛登录web测试用例", 1)
    print(e)
